# 简单生成 gif动画图片

from PIL import Image
import os

# 设置图片文件夹路径和输出 GIF 的保存路径
# image_folder = "202412\\花洒分类\\result_2\\result_0_label"  # 替换为存储图片的文件夹路径
# output_gif = "202412\\花洒分类\\result_2\\result_0_gif\\cluster.gif"  # 输出的 GIF 文件名

image_folder = "202412\\花洒分类\\result_2\\result_0"
output_gif = "202412\\花洒分类\\result_2\\result_0_gif\\original.gif"

# 获取图片文件列表，确保按顺序排列（可以根据文件名排序）
image_files = sorted([f for f in os.listdir(image_folder) if f.endswith(('png', 'jpg', 'jpeg'))])

# 读取图片并保存为 GIF
frames = []
for image_file in image_files:
    image_path = os.path.join(image_folder, image_file)
    img = Image.open(image_path)
    frames.append(img)

# 保存为 GIF
frames[0].save(
    output_gif,
    save_all=True,
    append_images=frames[1:],  # 附加其他帧
    optimize=True,
    duration=100,  # 每帧显示时长（毫秒）
    loop=0  # 循环次数，0 表示无限循环
)

print(f"GIF 动画已保存为 {output_gif}")
